cd ../../

gpus=$1
ids_bsz=24
pretrained_prefix="['encoder_st.encoder.rnn.']"
desirable_prefix="['rnn.']"
model_name="FrameATST/SimclrF_ByolaT_Mask0.4_MixupT_Fw1.01.0_FrzCnnF_LossS1.0W0.5O0.5"
echo "Run baseline with first 3-layer CNN initialized by Sup/SelfSL/Finetune results."
python sed_main.py   \
    --gpus "${gpus}," \
    --exp_name "sed/init/${model_name}/R_semi_3CfR_freeze_Self/" \
    --batch_sizes "[0, 12, 12, 24, 0]" \
    --pretrained_ckpt "/home/shaonian/SED/sssl_sed/codes/exp/semi/FrameATST/SimclrF_ByolaT_Mask0.4_MixupT_Fw1.01.0_FrzCnnF_LossS1.0W0.5O0.5/o96_i24/version_0/epoch=153-obj_metric=1.336.ckpt" \
    --pretrained_prefix ${pretrained_prefix} \
    --desirable_prefix ${desirable_prefix} \
    --freeze_cnn_layer "[-1]" 


pretrained_prefix="['encoder_st.encoder.cnn.cnn.conv3.',"` 
                  `"'encoder_st.encoder.cnn.cnn.batchnorm3.',"` 
                  `"'encoder_st.encoder.cnn.cnn.cg3.',"` 
                  `"'encoder_st.encoder.cnn.cnn.conv4.',"` 
                  `"'encoder_st.encoder.cnn.cnn.batchnorm4.',"` 
                  `"'encoder_st.encoder.cnn.cnn.cg4.',"` 
                  `"'encoder_st.encoder.cnn.cnn.conv5.',"` 
                  `"'encoder_st.encoder.cnn.cnn.batchnorm5.',"` 
                  `"'encoder_st.encoder.cnn.cnn.cg5.',"`
                  `"'encoder_st.encoder.cnn.cnn.conv6.',"`
                  `"'encoder_st.encoder.cnn.cnn.batchnorm6.',"` 
                  `"'encoder_st.encoder.cnn.cnn.cg6.']"
desirable_prefix="['cnn.cnn.conv3.',"` 
                `"'cnn.cnn.batchnorm3.',"` 
                `"'cnn.cnn.cg3.',"` 
                `"'cnn.cnn.conv4.',"` 
                `"'cnn.cnn.batchnorm4.',"` 
                `"'cnn.cnn.cg4.',"` 
                `"'cnn.cnn.conv5.',"` 
                `"'cnn.cnn.batchnorm5.',"` 
                `"'cnn.cnn.cg5.',"`
                `"'cnn.cnn.conv6.',"` 
                `"'cnn.cnn.batchnorm6.',"` 
                `"'cnn.cnn.cg6.']"

python sed_main.py   \
    --gpus "${gpus}," \
    --exp_name "sed/init/${model_name}/L4C_semi_3CfR_freeze_Self/" \
    --batch_sizes "[0, 12, 12, 24, 0]" \
    --pretrained_ckpt "/home/shaonian/SED/sssl_sed/codes/exp/semi/FrameATST/SimclrF_ByolaT_Mask0.4_MixupT_Fw1.01.0_FrzCnnF_LossS1.0W0.5O0.5/o96_i24/version_0/epoch=153-obj_metric=1.336.ckpt" \
    --pretrained_prefix ${pretrained_prefix} \
    --desirable_prefix ${desirable_prefix} \
    --freeze_cnn_layer "[3, 4, 5, 6]"


pretrained_prefix="['encoder_st.encoder.cnn.cnn.conv0.',"` 
                  `"'encoder_st.encoder.cnn.cnn.batchnorm0.',"` 
                  `"'encoder_st.encoder.cnn.cnn.cg0.',"` 
                  `"'encoder_st.encoder.cnn.cnn.conv1.',"` 
                  `"'encoder_st.encoder.cnn.cnn.batchnorm1.',"` 
                  `"'encoder_st.encoder.cnn.cnn.cg1.',"` 
                  `"'encoder_st.encoder.cnn.cnn.conv2.',"` 
                  `"'encoder_st.encoder.cnn.cnn.batchnorm2.',"` 
                  `"'encoder_st.encoder.cnn.cnn.cg2.']"
desirable_prefix="['cnn.cnn.conv0.',"` 
                `"'cnn.cnn.batchnorm0.',"` 
                `"'cnn.cnn.cg0.',"` 
                `"'cnn.cnn.conv1.',"` 
                `"'cnn.cnn.batchnorm1.',"` 
                `"'cnn.cnn.cg1.',"` 
                `"'cnn.cnn.conv2.',"` 
                `"'cnn.cnn.batchnorm2.',"` 
                `"'cnn.cnn.cg2.']"

python sed_main.py   \
    --gpus "${gpus}," \
    --exp_name "sed/init/${model_name}/F3C_semi_3CfR_freeze_Self/" \
    --batch_sizes "[0, 12, 12, 24, 0]" \
    --pretrained_ckpt "/home/shaonian/SED/sssl_sed/codes/exp/semi/FrameATST/SimclrF_ByolaT_Mask0.4_MixupT_Fw1.01.0_FrzCnnF_LossS1.0W0.5O0.5/o96_i24/version_0/epoch=153-obj_metric=1.336.ckpt" \
    --pretrained_prefix ${pretrained_prefix} \
    --desirable_prefix ${desirable_prefix} \
    --freeze_cnn_layer "[0, 1, 2]"